from BaseNN import nn
model = nn()
checkpoint = 'checkpoints/20240522/basenn.pth'
out_file='model/20240522.onnx'
model.convert(checkpoint=checkpoint, out_file=out_file)